tensor([[0.2600, 0.2600, 0.2600, 0.2600, 0.2600],
        [0.7400, 0.7400, 0.7400, 0.7400, 0.7400]], grad_fn=<RoundBackward1>)